#业务层实现登录、注册
import orm
def login():
    path=r'users.json'   # 引用json文件
    tag=0
    name=input("请输入用户名：")
    pwd=input("请输入密码：")
    user_list=orm.read_data(path) #获取json文件的数据
    for user in user_list:
        if name == user['username'] and pwd==user['password']: #对比json文件中的数据
            tag=1
            print('登录成功！')
            break
        else:
            print("用户名或密码错误！")
            break
    return tag

def register():
    path=r'users.json'   # 引用json文件
    name=input("请输入注册的用户名：")
    pwd=input("请输入密码：")
    new_user={'username':name,'password':pwd}
    user_list=orm.read_data(path) #获取json文件的数据
    user_list.append(new_user) #写入数据
    orm.write_data(user_list,path)  #将数据写入文件中
    print("注册成功")

if __name__ == '__main__':
    # print(assert())
    register()
